# https://github.com/codekansas/keras-language-modeling/blob/master/attention_lstm.py